# 使用官方 PyTorch GPU 镜像（已包含 PyTorch + CUDA 11.8 + Python 等）
FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime

# 设置工作目录
WORKDIR /app

# 拷贝你的项目代码
COPY ./ ./


# 可选：切换回代码目录
WORKDIR /app

# 可选：设置默认启动命令（比如运行 train.py）
#CMD ["python", "code/app_server.py"]